import timm
from pprint import pprint
import torch
from time import time
if __name__=='__main__':
    # model_names=timm.list_models(pretrained=True)
    # pprint(model_names)
    model=timm.create_model('regnetx_002',pretrained=False,num_classes=10)
    model=model.cuda()
    x=torch.rand((1,3,224,224),device='cuda')
    for i in range(100):
        with torch.no_grad():
            t1=time()
            y=model(x)
            t2=time()
            print(t2-t1)